Was going to do unit testing, but need to do more exploratory testing.
In [1]:
cd ..
In [6]:
%run train.py
In [3]:
%pdb